package 牛客网在线编程题库;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 题目描述
 * 给出一个整数数组，请在数组中找出两个加起来等于目标值的数，
 * 你给出的函数twoSum 需要返回这两个数字的下标（index1，index2），需要满足 index1 小于index2.。注意：下标是从1开始的
 * 假设给出的数组中只存在唯一解
 * 例如：
 * 给出的数组为 {20, 70, 110, 150},目标值为90
 * 输出 index1=1, index2=2
*
 * 示例1
 * 输入
 * 复制
 * [3,2,4],6
 * 返回值
 * 复制
 * [2,3]
 */

public class _两数之和 {
    public static void main(String[] args) {
        // 结果1 2
        int[] numbers = {20, 70, 110, 150};
        int target = 90;
        int[] result = twoSum(numbers, target);
        System.out.println(Arrays.toString(result));
    }

    public static int[] twoSum (int[] numbers, int target) {
        // write code here
        // 若数组为空，则直接返回
        if (numbers == null) {
            return new int[2];
        }
        // 定义哈希表，键存储数组的元素值，值存储元素值对应的下标
        Map<Integer, Integer> map = new HashMap<>();
        // 遍历
        for (int i = 0 ; i < numbers.length ; i++) {
            // 记录当前数
            int rest = numbers[i];
            // 检索哈希表中有无当前数对应的另一个数
            if (map.containsKey(target - rest)) {
                // 如果有则直接返回
                return new int[]{map.get(target - rest) + 1, i + 1};
            }
            // 如果没有就加入哈希表
            map.put(rest, i);
        }
        return new int[2];
    }
}
